from smolagents import CodeAgent, LiteLLMModel, WebSearchTool, VisitWebpageTool
from tools.tools import CampusServiceTools

model = LiteLLMModel(model_id="anthropic/claude-3-5-sonnet-latest", api_key="YOUR_ANTHROPIC_API_KEY") # Could use 'gpt-4o'
agent = CodeAgent(
    tools=[
        "life_service",
        "medical_service",
        "transport_service",
        "service_orgs",
        "school_news",
        "school_profile",
        "president_message",
        "academic_events",
        "school_song",
    ],
    model=model,
    add_base_tools=True,
)

campus_tools = CampusServiceTools()

agent.run(
    "请访问 https://www.nfu.edu.cn/xyfw/syxx/shfw.htm 告诉我餐厅营业时间"
)

